package com.kun.utils;

import com.github.pagehelper.PageInfo;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;

import java.util.HashMap;
import java.util.List;

/**
 * 操作消息提醒
 */
public class AjaxResult extends HashMap<String, Object> {

    /**
     * 状态码
     */
    public static final String CODE_TAG = "code";

    /**
     * 返回内容
     */
    public static final String MSG_TAG = "msg";

    /**
     * 数据对象
     */
    public static final String DATA_TAG = "data";

    /**
     * 数据数量
     */
    public static final String COUNT_TAG = "count";

    /**
     * 初始化一个新创建的 AjaxResult 对象，使其表示一个空消息。
     */
    public AjaxResult() {
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     */
    public AjaxResult(int code, String msg, Object data) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
        if (data != null) {
            super.put(DATA_TAG, data);
        }
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     */
    public AjaxResult(int code, String msg) {
        super.put(CODE_TAG, code);
        super.put(MSG_TAG, msg);
    }

    /**
     * 返回成功消息
     */
    public static ResponseEntity<AjaxResult> success() {
        AjaxResult ajaxResult = new AjaxResult(HttpStatus.OK.value(), "操作成功", null);
        return ResponseEntity.ok(ajaxResult);
    }

    /**
     * 返回成功数据
     */
    public static ResponseEntity<AjaxResult> success(Object data) {
        AjaxResult success = AjaxResult.success("操作成功", data);
        return new ResponseEntity<>(success, HttpStatus.OK);
    }

    /**
     * 返回成功数据
     */
    public static ResponseEntity<AjaxResult> success(String msg) {
        AjaxResult ajaxResult = new AjaxResult(HttpStatus.OK.value(), msg);
        return new ResponseEntity<>(ajaxResult, HttpStatus.OK);
    }

//    /**
//     * 返回成功消息
//     *
//     * @param msg 返回内容
//     * @return 成功消息
//     */
//    public static AjaxResult success(String msg) {
//        return AjaxResult.success(msg, null);
//    }

    /**
     * 返回成功消息
     */
    public static AjaxResult success(String msg, Object data) {
        return new AjaxResult(HttpStatus.OK.value(), msg, data);
    }

    /**
     * 返回成功消息
     */
    public static ResponseEntity<AjaxResult> successEntity(HttpStatus httpStatus, String msg, Object data) {
        AjaxResult ajaxResult = new AjaxResult(httpStatus.value(), msg, data);
        return new ResponseEntity<>(ajaxResult, httpStatus);
    }

    /**
     * 返回警告消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static AjaxResult warn(String msg) {
        return AjaxResult.warn(msg, null);
    }

    /**
     * 返回警告消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static AjaxResult warn(String msg, Object data) {
        return new AjaxResult(HttpStatus.BAD_GATEWAY.value(), msg);
    }

    /**
     * 返回错误消息
     */
    public static ResponseEntity<AjaxResult> error() {
        return AjaxResult.error("操作失败");
    }

    /**
     * 返回错误消息
     */
    public static ResponseEntity<AjaxResult> error(String msg) {
        return AjaxResult.error(msg, null);
    }

    /**
     * 返回错误消息
     */
    public static ResponseEntity<AjaxResult> error(String msg, Object data) {
        AjaxResult ajaxResult = new AjaxResult(HttpStatus.INTERNAL_SERVER_ERROR.value(), msg, data);
        return new ResponseEntity<>(ajaxResult, HttpStatus.INTERNAL_SERVER_ERROR);
    }

    /**
     * 返回错误消息
     */
    public static ResponseEntity<AjaxResult> error(HttpStatus httpStatus, String msg) {
        AjaxResult ajaxResult = new AjaxResult(httpStatus.value(), msg);
        return new ResponseEntity<>(ajaxResult, httpStatus);
    }

    /**
     * 返回表格分页数据
     */
    public static AjaxResult getDataTable(List<?> list) {
        AjaxResult success = AjaxResult.success("操作成功", list);
        PageInfo<?> pageInfo = PageInfo.of(list);
        success.put(COUNT_TAG, pageInfo.getTotal());
        return success;
    }

//    /**
//     * 是否为成功消息
//     *
//     * @return 结果
//     */
//    public boolean isSuccess() {
//        return Objects.equals(HttpStatus.SUCCESS, this.get(CODE_TAG));
//    }
//
//    /**
//     * 是否为警告消息
//     *
//     * @return 结果
//     */
//    public boolean isWarn() {
//        return Objects.equals(HttpStatus.WARN, this.get(CODE_TAG));
//    }
//
//    /**
//     * 是否为错误消息
//     *
//     * @return 结果
//     */
//    public boolean isError() {
//        return Objects.equals(HttpStatus.ERROR, this.get(CODE_TAG));
//    }

    /**
     * 方便链式调用
     *
     * @param key   键
     * @param value 值
     * @return 数据对象
     */
    @Override
    public AjaxResult put(String key, Object value) {
        super.put(key, value);
        return this;
    }


}
